<template>
  <div id="sql">
    <div class="sql-main">
      <!-- 选择区 -->
      <el-card class="selection-area">
        <el-form :inline="true" :model="form" class="demo-form-inline">
          <el-form-item label="数据库类型">
            <el-select v-model="form.dataType" placeholder="请选择数据库类型" @change="dataTypeChange" clearable>
              <el-option
                v-for="item in dataTypeOptions"
                :key="item.value"
                :label="item.label"
                :value="item.value">
              </el-option>
            </el-select>
          </el-form-item>
          <el-form-item label="系统名称">
            <el-select v-model="form.name" placeholder="请选择系统名称" clearable @click.native="nameClick">
              <el-option
                v-for="item in nameOptions"
                :key="item.value"
                :label="item.label"
                :value="item.value">
              </el-option>
            </el-select>
          </el-form-item>
          <el-form-item label="数据库名">
            <el-select v-model="form.dataName" placeholder="请选择数据库名" clearable @change="dataNameChange" @click.native="dataNameClick">
              <el-option
                v-for="item in dataNameOptions"
                :key="item.value"
                :label="item.label"
                :value="item.value">
              </el-option>
            </el-select>
          </el-form-item>
          <el-form-item>
            <div class="run" @click="executeQuery">
              <svg-icon icon-class="run" />
              <span>{{ runText }}</span>
            </div>
          </el-form-item>
        </el-form>
      </el-card>
      <!-- sql输入区 -->
      <el-card class="sql-input">
        <el-input type="textarea" v-model="sql" @select.native="selectSql" @mouseup.native="mouseupSql" @blur="blurSql" @keyup.native="keyupSql"></el-input>
      </el-card>
      <!-- 查询结果 -->
      <!-- 给el-tabs设置默认选中（value与el-tab-pane标签中的name相等即为默认选中name页），如果有错误信息，默认选中错误信息标签页，如果没有，默认选中tab-0标签页，在v-for中给每个el-tab-pane加上name属性 -->
      <el-tabs type="border-card" :value="errorMsg == null ? 'tab-0' : 'error'" v-show="result.length != 0 || errorMsg != null" @tab-click="tabClick">
        <el-tab-pane label="错误信息" v-if="errorMsg != null" class="errTab" name="error">
          <div>{{ errorMsg }}</div>
        </el-tab-pane>
        <el-tab-pane :label="result.length == 1 ? '结果' : '结果' + (index + 1)" v-for="(item, index) in result" :key="index + 'tab'" :name="'tab-' + index">
          <!-- 导出按钮 -->
          <el-button class="export" @click="exportRes(item.sql)">导出结果</el-button>
          <!-- 循环获取表数据 -->
          <el-table :data="item.resultList" height="260" border style="width: 100%;">
            <!-- 循环获取表头 -->
            <el-table-column :prop="title" :label="title" width="180" v-for="(title, i) in item.nameList" :key="i + 'column'">
            </el-table-column>
          </el-table>
        </el-tab-pane>
      </el-tabs>
      <!-- <el-button class="export" v-show="result.length != 0 || errorMsg != null">导出结果</el-button> -->
    </div>
    <div class="sql-tree">
      <el-card>
        <el-tree :data="treeList" :props="defaultProps" @node-click="handleNodeClick"></el-tree>
      </el-card>
    </div>

    <el-dialog
      title="请选择导出文件格式"
      :visible.sync="dialogVisible"
      width="30%"
      center>
      <el-radio-group v-model="exportType">
        <el-radio :label="item.type" v-for="item in fileTypeOptions" :key="item.type">{{ item.desc }}</el-radio>
      </el-radio-group>
      <span slot="footer" class="dialog-footer">
        <el-button type="primary" @click="confirmExport">确 定</el-button>
        <el-button @click="dialogVisible = false">取 消</el-button>
      </span>
    </el-dialog>
  </div>
</template>

<script>
  import common from '@/api/common'
  import connect from '@/api/connect'
  import sql from '@/api/sql'
  import dataQuery from '@/api/dataQuery'
  export default {
    name: 'SQL',
    data() {
      return {
        form: {
          dataType: 'mysql',
          name: 'localhost',
          dataName: 'datasource-manage'
        },
        // 数据库类型下拉选择器
        dataTypeOptions: [],
        // 系统名称下拉选择器
        nameOptions: [],
        // 数据库名称下拉选择器
        dataNameOptions: [],
        // 系统名称下类选择器是否禁用
        nameDisabled: true,
        // 数据库名称下拉选择器是否禁用
        dataNameDisabled: true,
        // 是否是原来的数据源
        isOriginal: true,
        // sql语句
        sql: null,
        // 选中的sql
        currentSql: null,
        // isSelect: false,
        // sql查询结果
        result: [],
        errorMsg: null,
        // 导出弹出框是否显示
        dialogVisible: false,
        // 导出文件类型
        exportType: null,
        // 文件类型
        fileTypeOptions: [],
        // 要导出的数据对应的sql（sql是key，数据存在redis里，需要用key去取）
        exportSql: null,
        // textareaValue: '<textarea autocomplete="off" class="el-textarea__inner" style="min-height: 32.6px;"></textarea>'
        treeList:  [{
          label: '表',
          children: []
        }, {
          label: '视图',
          children: []
        }],
        defaultProps: {
          children: 'children',
          label: 'label'
        }
      }
    },
    // 计算属性
    computed: {
      runText() {
        return this.currentSql == '' || this.currentSql == null ? '运行' : '运行已选择的';
      }
    },
    // 监听器，监听下拉列表
    watch: {
      'form.dataType'(newVal, oldVal) {
        this.form.name = null
        if (newVal) {
          this.nameDisabled = false
        } else {
          this.nameDisabled = true
        }
      },
      'form.name'(newVal, oldVal) {
        this.form.dataName = null
        if (newVal) {
          this.dataNameDisabled = false
        } else {
          this.dataNameDisabled = true
        }
      }
    },
    methods: {
      // 根据数据库类型查询系统名称
      getNameByDataType() {
        if (!this.form.dataType) {
          return;
        }
        connect.getNameByDataType(this.form.dataType).then(res => {
          // 先将原来的数组清空
          this.nameOptions = []
          res.data.map(item => {
            this.nameOptions.push({
              value: item,
              label: item
            })
          })
        })
      },
      // 根据数据库类型和系统名称查询数据库名
      getDataNameByDataTypeAndName() {
        if (!this.form.name) {
          return;
        }
        connect.getDataNameByDataTypeAndName(this.form.dataType, this.form.name).then(res => {
          // 先将原来的数组清空
          this.dataNameOptions = []
          res.data.map(item => {
            this.dataNameOptions.push({
              value: item,
              label: item
            })
          })
        })
      },
      // 切换回原来的数据源
      switchOriginal() {
        if (!this.isOriginal) {
          connect.switchDataSource({
            "name": "localhost",
            "url": "127.0.0.1",
            "port": "3306",
            "dataName": "datasource-manage",
            "username": "root",
            "password": "123456",
            "dataType": "mysql"
          }).then(res => {
            if (res.code == 200) {
              this.isOriginal = true
              console.log("switchOriginal")
            }
          })
        }
      },
      // 数据库类型发生改变，切换回原来的数据源
      dataTypeChange() {
        this.switchOriginal()
      },
      // 系统名称下拉选择器点击事件
      nameClick() {
        if (this.isOriginal) {
          this.getNameByDataType()
        } else {
          this.switchOriginal()
        }
      },
      // 数据库名称下拉选择器点击事件
      dataNameClick() {
        if (this.isOriginal) {
          this.getDataNameByDataTypeAndName()
        } else {
          this.switchOriginal()
        }
      },
      // 当选择数据库后，切换数据源
      dataNameChange() {
        console.log(this.form)
        // 查询数据源
        connect.getConn(this.form).then(res => {
          if (res.code == 200) {
            console.log(res)
            // 切换数据源
            connect.switchDataSource(res.data).then(resp => {
              console.log(resp)
              if (resp.code == 200) {
                this.isOriginal = false
                this.showTables()
                this.showViews()
              }
            })
          }
        })
      },
      // 执行sql语句
      executeQuery() {
        console.log(this.form)
        if (this.form.dataType == null || this.form.name == null || this.form.dataName == null) {
          this.$message.error("请先选择数据源");
          return;
        }
        // console.log(this.sql, this.currentSql)
        // 执行前现将错误信息清空
        this.errorMsg = null
        // 要传递给后端的参数
        let param;
        // 如果没有选择，就执行所有的；如果选择了，就执行选择的
        if (!this.currentSql) {
          param = this.sql
        } else {
          param = this.currentSql
        }
        sql.executeQuery(param).then(res => {
          if (res.code != 200) {
            this.errorMsg = res.msg
          }
          this.result = res.data
        })
      },
      // 选中sql，调用input的原生时间，@click需要用native修饰
      selectSql(e) {
        // // 获取选中字符串的起始下标、结束下标与输入框的值
        // let start = e.target.selectionStart
        // let end = e.target.selectionEnd
        // let val = e.target.value
        // let selectVal = ''
        // // 防止没有选中
        // if (val && end != 0) {
        //   selectVal = val.slice(start, end)
        // }

        // const highlightedText = `<mark>${selectVal}</mark>`; // 使用 <mark> 标签包裹选中的文本
        // const newText = val.slice(0, start) + highlightedText + val.slice(end);
        // this.textareaValue = `<textarea autocomplete="off" class="el-textarea__inner" style="min-height: 32.6px;">${newText}</textarea>`

        let selectedText = '';
        if (window.getSelection().toString().length > 0) {
          selectedText = window.getSelection().toString();
        } else {
          selectedText = '';
        }
        this.currentSql = selectedText
      },
      // 鼠标点击取消选择
      mouseupSql() {
        if (this.currentSql) {
          this.currentSql = null
        }
      },
      // 文本域失去焦点取消选择
      blurSql() {
        // 定时器，避免失去焦点时直接清空选中sql，这样当选中sql后点击运行按钮是有效果的
        if (this.currentSql) {
          setTimeout(() => {
            this.currentSql = null
          }, 500);
        }
      },
      // 键盘按下取消选择
      keyupSql(e) {
        // 判断是否使用键盘选中
        if (this.currentSql && !['ArrowUP', 'ArrowDown', 'ArrowLeft', 'ArrowRight', 'Shift', 'Home', 'End'].includes(e.key)) {
          this.currentSql = null
        }
      },
      tabClick(tab, event) {
        console.log(tab)
        console.log(event)
      },
      // 导出结果按钮点击事件
      exportRes(res) {
        this.exportSql = res
        this.dialogVisible = true
      },
      // 确定导出
      confirmExport() {
        sql.exportResult(this.exportType, this.exportSql).then(res => {
          // 获取后端响应头中的文件名
          const contentDisposition = res.headers['content-disposition'];
          const match = contentDisposition.match(/filename[^;=\n]*=((['"]).*?\2|[^;\n]*)/);
          let fileName = 'exported_file'; // 默认文件名
          if (match && match[1]) {
            fileName = decodeURIComponent(match[1]);
          }
          // 将文件内容封装为 Blob 对象
          const blob = new Blob([res.data], { type: res.headers['content-type'] });
          // 创建临时 URL，并模拟点击下载
          const url = URL.createObjectURL(blob);
          const a = document.createElement('a');
          a.href = url;
          a.download = fileName;
          a.click();
          // 释放临时 URL
          URL.revokeObjectURL(url);
        })
      },
      handleNodeClick(data) {
        console.log(data);
      },
      // 查询当前库下所有表，不包括视图
      showTables() {
        this.treeList[0].children = []
        dataQuery.showTables(this.form.dataType, this.form.dataName).then(res => {
          console.log(res)
          res.data.forEach(item => {
            this.treeList[0].children.push({
              label: item.TABLE_NAME
            })
          })
        })
      },
      // 查询当前库下所有视图
      showViews() {
        this.treeList[1].children = []
        dataQuery.showViews(this.form.dataType, this.form.dataName).then(res => {
          // console.log(res)
          res.data.forEach(item => {
            this.treeList[1].children.push({
              label: item.TABLE_NAME
            })
          })
        })
      },
      // 页面刷新时执行此方法
      handleBeforeUnload() {
        // 在页面进行刷新时判断是否切换数据源
        // this.switchOriginal()
      }
    },
    mounted() {
      // 监听 beforeunload 事件
      window.addEventListener('beforeunload', this.handleBeforeUnload)
      // 加载数据库类型选择器数据
      common.getDataType().then(res => {
        this.dataTypeOptions = res.data;
      })
      // 获取文件类型
      common.getFileType().then(res => {
        this.fileTypeOptions = res.data
      })
      this.showTables()
      this.showViews()
    },
    beforeDestroy() {
      // 在组件销毁前移除 beforeunload 事件监听
      window.removeEventListener('beforeunload', this.handleBeforeUnload)
    },
    // 离开当前页执行
    beforeRouteLeave (to, from, next) {
      // 在离开当前页面时调用switchOriginal进行判断当前数据源是否是初始数据源，如果不是，切换回来
      this.switchOriginal()
      next()
    },
    beforeRouteEnter (to, from, next) {
      // ...进入当前页时切换数据源
      connect.switchDataSource({
        "name": "localhost",
        "url": "127.0.0.1",
        "port": "3306",
        "dataName": "datasource-manage",
        "username": "root",
        "password": "123456",
        "dataType": "mysql"
      })
      next()
    }
  }
</script>

<style lang="scss" scoped>
  #sql {
    position: relative;
    height: 100%;
    display: flex;
    // 查询区域
    .sql-main {
      flex: 1;
      min-width: 1096px;
      .el-tabs {
        // 父元素宽度-右边树结构宽度200-树结构右边距10
        width: calc(100% - 210px);
        min-width: 1096px;
        height: 300px;
        position: absolute;
        // position: relative;
        bottom: 0;
        .errTab {
          display: flex;
          justify-content: center; 
          align-items: center;
          height: 260px;
        }
        .el-tabs__content {
          .export {
            position: absolute;
            top: -40px;
            right: 0;
            z-index: 9999;
          }
        }
        
      }
      // @media (min-width: 1196px) {
      //   .export {
      //     position: absolute;
      //     bottom: 260px;
      //     right: 210px;
      //   }
      // }
      // @media (max-width: 1196px) {
      //   .export {
      //     position: absolute;
      //     bottom: 260px;
      //     right: 0;
      //   }
      // }
    }
    // 右侧树结构区域
    .sql-tree {
      width: 200px;
      min-width: 200px;
      margin-left: 10px;
      .el-card {
        height: 100%;
      }
    }
    .el-dialog {
      
      .el-radio {
        display: block;
        margin: 15px 0;
      }
      .el-button {
        margin: 0 15px;
      }
    }
  }
  .selection-area.el-card {
    .el-form-item {
      margin: 0 30px 0 0;
      .run {
        display: flex;
        align-items: center;
        cursor: pointer;
        &:hover {
          background: rgba(216,230,242, 0.8);
        }
        .svg-icon {
          width: 30px;
          height: 30px;
        }
      }
    }
  }
  
</style>

<style scoped>
  .selection-area >>> .el-input__inner {
    width: 160px;
  }
  .sql-input >>> .el-textarea__inner {
    height: 350px;
  }
  >>> .el-tabs--border-card>.el-tabs__content {
    padding: 0;
  }
  /* 导出按钮相对tab内容定位，但是他默认overflow:hidden，会隐藏按钮，因此改为visible */
  >>>.el-tabs__content {
    overflow: visible;
  }
  >>>.el-dialog__body {
      padding: 15px 25px;
    }
</style>